from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain_community.llms import OpenAI  # 或 from langchain_openai import ChatOpenAI
from langchain_openai import ChatOpenAI

# 1. 创建模板
prompt = PromptTemplate(
    input_variables=["product"],
    template="为一家生产{product}的公司起一个好听的名字，并给出一个 slogan。",
)

# 2. 创建模型
#llm = OpenAI(temperature=0.7) # 或者 ChatOpenAI(model="gpt-3.5-turbo")
llm = ChatOpenAI(
            model="deepseek-chat",
            api_key="sk-079f9ad2ad3f457ebd6e6eb90f56fb53",
            base_url="https://api.deepseek.com/v1",  # DeepSeek API 地址
            temperature=0.7
        )
# 3. 创建链
chain = LLMChain(llm=llm, prompt=prompt)

# 4. 运行链
result = chain.run("环保可降解的咖啡杯")
print(result)
# 输出可能:
# 公司名: 绿源杯业
# Slogan: 品味咖啡，守护地球每一杯。

